#ifndef __IANGELS_OP_RESHAPE_H__
#define __IANGELS_OP_RESHAPE_H__

#include "op/op.h"

namespace iAngels {

class Reshape: public OpBase {
public:
    Reshape(const Config& config): OpBase(config) {}
    ~Reshape() {}

    Ret operator()(const std::vector<LazyDynamicTensor>& inputs,
        std::vector<LazyDynamicTensor>& outputs, bool inplace=false);
private:
    std::vector<int> reshape_dims;
}; // class Reshape

} // namespace iAngels
#endif // __IANGELS_OP_RESHAPE_H__